/*
 * SPDX-FileCopyrightText: 2022 Espressif Systems (Shanghai) CO LTD
 *
 * SPDX-License-Identifier: Unlicense OR CC0-1.0
 */

#include <string.h>
#include <stdio.h>
#include <stddef.h>
#include <stdbool.h>
#include "inno_connect.h"
#include "example_common_private.h"
#include "esp_wifi.h"
#include "esp_log.h"
#include "esp_console.h"
#include "argtable3/argtable3.h"

static const char *TAG = "inno_console";

typedef struct {
	struct arg_str *ssid;
	struct arg_str *password;
	struct arg_int *channel;
	struct arg_end *end;
} wifi_connect_args_t;
static wifi_connect_args_t connect_args;

static int cmd_do_wifi_connect(int argc, char **argv)
{
	int nerrors = arg_parse(argc, argv, (void**) &connect_args);

	if (nerrors != 0) {
		arg_print_errors(stderr, connect_args.end, argv[0]);
		return 1;
	}

	wifi_config_t wifi_config = {
			.sta = {
					.scan_method = WIFI_ALL_CHANNEL_SCAN,
					.sort_method = WIFI_CONNECT_AP_BY_SIGNAL,
			},
	};
	if (connect_args.channel->count > 0) {
		wifi_config.sta.channel = (uint8_t) (connect_args.channel->ival[0]);
	}
	const char *ssid = connect_args.ssid->sval[0];
	const char *pass = connect_args.password->sval[0];
	strlcpy((char*) wifi_config.sta.ssid, ssid, sizeof(wifi_config.sta.ssid));
	if (pass) {
		strlcpy((char*) wifi_config.sta.password, pass,
				sizeof(wifi_config.sta.password));
	}
	inno_wifi_sta_do_connect(wifi_config, false);
	return 0;
}

static int cmd_do_wifi_disconnect(int argc, char **argv)
{
	example_wifi_sta_do_disconnect();
	return 0;
}

void example_register_wifi_connect_commands(void)
{
	ESP_LOGI(TAG, "Registering WiFi connect commands.");
	connect_wifi_start();

	connect_args.ssid = arg_str1(NULL, NULL, "<ssid>", "SSID of AP");
	connect_args.password = arg_str0(NULL, NULL, "<pass>", "password of AP");
	connect_args.channel = arg_int0("n", "channel", "<channel>",
			"channel of AP");
	connect_args.end = arg_end(2);
	const esp_console_cmd_t wifi_connect_cmd = {
			.command = "wifi_connect",
			.help = "WiFi is station mode, join specified soft-AP",
			.hint = NULL,
			.func = &cmd_do_wifi_connect,
			.argtable = &connect_args
	};
	ESP_ERROR_CHECK(esp_console_cmd_register(&wifi_connect_cmd));

	const esp_console_cmd_t wifi_disconnect_cmd = {
			.command = "wifi_disconnect",
			.help = "Do wifi disconnect",
			.hint = NULL,
			.func = &cmd_do_wifi_disconnect,
	};
	ESP_ERROR_CHECK(esp_console_cmd_register(&wifi_disconnect_cmd));
}
